import cv2
id = input("请输入采集对象的序号：")

cv2.namedWindow('test',cv2.WINDOW_NORMAL)
cv2.resizeWindow('test',300,300)  #设置窗口大小


# 创建VideoCapture对象，打开摄像头（内接为0，外接为1）
# 实时录制视频（现有视频写入参数：“文件名”）pip install opencv-pythonpip install opencv-python
cap = cv2.VideoCapture(0)
# 设置摄像头参数（3是宽，4是高）
cap.set(3, 640)
cap.set(4, 488)

while cap.isOpened():
    ret, frame = cap.read()  # 将视频变帧
    if ret == True:
        # print("frame.shape:{}".format(frame.shape))  # 输出当前帧的高度、宽度和深度
        cv2.imshow("test", frame)  # 显示捕获的帧
        cv2.resizeWindow('test', 300, 300)  # 重新设置窗口大小(否则大小会变化)
        key = cv2.waitKey(10)
        if key & 0xFF == 27:  # 代表ESC键推出(排除其他键的干扰，永远后八位)
            break
        elif key == ord('s') or key == ord('S'):  # 自定义按’s‘键代表保存图像并退出
            save_path = 'C:/Users/shouw/Desktop/tmp/known_pictures/'                # capturesou
            cv2.imwrite(save_path + 'person' + str(id) + '.jpg', frame)
            break
    else:
        break
cap.release()
cv2.destroyAllWindows()
